package com.ruoyi.plus.utils;

import com.ruoyi.common.utils.StringUtils;
import com.ruoyi.common.utils.spring.SpringUtils;
import lombok.SneakyThrows;
import lombok.extern.slf4j.Slf4j;
import org.apache.commons.codec.binary.Base64;
import org.springframework.context.annotation.Bean;
import org.springframework.data.redis.core.RedisTemplate;
import org.springframework.stereotype.Component;

import javax.crypto.Cipher;
import java.io.Serializable;
import java.security.*;
import java.security.interfaces.RSAPrivateKey;
import java.security.interfaces.RSAPublicKey;
import java.security.spec.PKCS8EncodedKeySpec;
import java.security.spec.X509EncodedKeySpec;

@Component
@Slf4j
public class RsaUtils {
    // Rsa 私钥 也可固定秘钥对 若依原写法(不安全)
    //public static String privateKeys = "";
    //private static String publicKeyStr = "";
    //private static String privateKeyStr = "";
    private static final RsaKeyPair rsaKeyPair = new RsaKeyPair();
    private static final String RSA_KEY_PAIR = "rsaKeyPair";
    //    @Autowired
//    private RedisTemplate<String, Object> redisTemplate;
    private static RedisTemplate redisTemplate = SpringUtils.getBean("redisTemplate");

    /**
     * 私钥解密
     *
     * @param text 待解密的文本
     * @return 解密后的文本
     */

    @SneakyThrows
    public static String decryptByPrivateKey(String text) {
        return decryptByPrivateKey(rsaKeyPair.getPrivateKey(), text);
    }

    /**
     * 公钥解密
     *
     * @param publicKeyString 公钥
     * @param text            待解密的信息
     * @return 解密后的文本
     */
    @SneakyThrows
    public static String decryptByPublicKey(String publicKeyString, String text) {
        X509EncodedKeySpec x509EncodedKeySpec = new X509EncodedKeySpec(Base64.decodeBase64(publicKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PublicKey publicKey = keyFactory.generatePublic(x509EncodedKeySpec);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.DECRYPT_MODE, publicKey);
        byte[] result = cipher.doFinal(Base64.decodeBase64(text));
        return new String(result);
    }

    /**
     * 私钥加密
     *
     * @param privateKeyString 私钥
     * @param text             待加密的信息
     * @return 加密后的文本
     */
    @SneakyThrows
    public static String encryptByPrivateKey(String privateKeyString, String text) {
        PKCS8EncodedKeySpec pkcs8EncodedKeySpec = new PKCS8EncodedKeySpec(Base64.decodeBase64(privateKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PrivateKey privateKey = keyFactory.generatePrivate(pkcs8EncodedKeySpec);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.ENCRYPT_MODE, privateKey);
        byte[] result = cipher.doFinal(text.getBytes());
        return Base64.encodeBase64String(result);
    }

    /**
     * 私钥解密
     *
     * @param privateKeyString 私钥
     * @param text             待解密的文本
     * @return 解密后的文本
     */
    @SneakyThrows
    public static String decryptByPrivateKey(String privateKeyString, String text) {
        PKCS8EncodedKeySpec pkcs8EncodedKeySpec5 = new PKCS8EncodedKeySpec(Base64.decodeBase64(privateKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PrivateKey privateKey = keyFactory.generatePrivate(pkcs8EncodedKeySpec5);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.DECRYPT_MODE, privateKey);
        byte[] result = cipher.doFinal(Base64.decodeBase64(text));
        return new String(result);
    }

    /**
     * 公钥加密
     *
     * @param publicKeyString 公钥
     * @param text            待加密的文本
     * @return 加密后的文本
     */
    @SneakyThrows
    public static String encryptByPublicKey(String publicKeyString, String text) {
        X509EncodedKeySpec x509EncodedKeySpec2 = new X509EncodedKeySpec(Base64.decodeBase64(publicKeyString));
        KeyFactory keyFactory = KeyFactory.getInstance("RSA");
        PublicKey publicKey = keyFactory.generatePublic(x509EncodedKeySpec2);
        Cipher cipher = Cipher.getInstance("RSA");
        cipher.init(Cipher.ENCRYPT_MODE, publicKey);
        byte[] result = cipher.doFinal(text.getBytes());
        return Base64.encodeBase64String(result);
    }

    public static String getPublicKey() {
        return rsaKeyPair.getPublicKey();
    }

    public static String getPrivateKey() {
        return rsaKeyPair.getPrivateKey();
    }

    public static RsaKeyPair rsaKeyPair() {
        return rsaKeyPair;
    }

    /**
     * 构建RSA密钥对
     */
    @Bean
    @SneakyThrows
    public void generateKeyPair() {
        RsaKeyPair rsaKeyPair1 = null;
        try {
            rsaKeyPair1 = (RsaKeyPair) redisTemplate.opsForValue().get(RSA_KEY_PAIR);
        } catch (Exception e) {
            log.error(e.getMessage());
        }
        if (rsaKeyPair1 == null || StringUtils.isEmpty(rsaKeyPair1.getPublicKey()) || StringUtils.isEmpty(rsaKeyPair1.getPrivateKey())) {
            KeyPairGenerator keyPairGenerator = KeyPairGenerator.getInstance("RSA");
            keyPairGenerator.initialize(1024);
            KeyPair keyPair = keyPairGenerator.generateKeyPair();
            RSAPublicKey rsaPublicKey = (RSAPublicKey) keyPair.getPublic();
            RSAPrivateKey rsaPrivateKey = (RSAPrivateKey) keyPair.getPrivate();
            String publicKeyString = Base64.encodeBase64String(rsaPublicKey.getEncoded());
            String privateKeyString = Base64.encodeBase64String(rsaPrivateKey.getEncoded());
            rsaKeyPair.setPrivateKey(privateKeyString);
            rsaKeyPair.setPublicKey(publicKeyString);
            //保存到redis
            redisTemplate.opsForValue().set(RSA_KEY_PAIR, rsaKeyPair);
        } else {
            rsaKeyPair.setPublicKey(rsaKeyPair1.getPublicKey());
            rsaKeyPair.setPrivateKey(rsaKeyPair1.getPrivateKey());
        }
    }

    /**
     * RSA密钥对对象
     */
    public static class RsaKeyPair implements Serializable {
        private String publicKey;
        private String privateKey;

        public RsaKeyPair() {

        }

        public RsaKeyPair(String publicKey, String privateKey) {
            this.publicKey = publicKey;
            this.privateKey = privateKey;
        }

        public String getPublicKey() {
            return publicKey;
        }

        public void setPublicKey(String publicKey) {
            this.publicKey = publicKey;
        }

        public String getPrivateKey() {
            return privateKey;
        }

        public void setPrivateKey(String privateKey) {
            this.privateKey = privateKey;
        }
    }
}
